from selenium import  webdriver
from time import sleep
import  unittest
from selenium.webdriver.support.wait import WebDriverWait
class baiDuTestCase(unittest.TestCase):
    #class前置条件
    @classmethod
    def setUpClass(cls) -> None:
        print("class前置条件")

    #class后置条件
    @classmethod
    def tearDownClass(cls) -> None:
        print("class后置条件")

    #每个用例方法之前运行
    def setUp(self):
        print("开始执行测试用例")
        url = "http://www.baidu.com"
        self.driver = webdriver.Chrome()
        self.driver.get(url)

    #测试用例的执行顺序是根据用例名称来顺序执行的
    def test_button(self):  #测试用例的名称要以test开头
        self.driver.find_element_by_id("kw").send_keys("虚竹")
        self.driver.find_element_by_id("su").click()
        sleep(2)
        self.driver.save_screenshot('./screenshot/baidu.png')

    #每个用例方法之后执行
    def tearDown(self):
        print("一条测试用例执行完毕")
        self.driver.quit() #退出浏览器

    if __name__ =='__main__':  #运行主函数
        unittest.main()

